package com.jsj.subway.ai.agent.rag;

import com.jsj.subway.ai.agent.rag.models.KnowledgeBase;
import com.jsj.subway.ai.largemodel.models.Result;
import okhttp3.MultipartBody;
import retrofit2.Call;
import retrofit2.http.Body;
import retrofit2.http.POST;
import retrofit2.http.Query;

public interface KbApi {

    @POST("/api/knowledge/create_kb")
    Call<Result.KbBase> createkb (@Body KnowledgeBase.CreateReq req);

    @POST("/api/knowledge/delete_knowledge")
    Call<Result.KbBase> deletekb (@Body KnowledgeBase.DeleteReq req);

    @POST("/api/knowledge/add_knowledge_file")
    Call<Result.KbBase> uploadKnowledgeBaseDocs(@Query("file_id") String file_id, @Body MultipartBody multipartBody);

    @POST("/api/knowledge/delete_knowledge_file")
    Call<Result.KbBase> downloadKnowledgeBaseDoc (@Body KnowledgeBase.DeleteDocReq req);

    @POST("/api/knowledge/query")
    Call<Result.KbBase> queryKnowledgeBase (@Body KnowledgeBase.QueryReq req);
}
